// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/public/test/content_browser_test.h"

#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/logging.h"
#include "base/message_loop/message_loop.h"
#include "base/path_service.h"
#include "base/run_loop.h"
#include "base/threading/thread_restrictions.h"
#include "build/build_config.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/common/content_paths.h"
#include "content/public/common/content_switches.h"
#include "content/public/common/url_constants.h"
#include "content/shell/browser/shell.h"
#include "content/shell/browser/shell_browser_context.h"
#include "content/shell/browser/shell_content_browser_client.h"
#include "content/shell/common/shell_switches.h"
#include "content/shell/renderer/layout_test/layout_test_content_renderer_client.h"
#include "content/test/test_content_client.h"

#if defined(OS_ANDROID)
#include "content/shell/app/shell_main_delegate.h"
#endif

#if defined(OS_MACOSX)
#include "base/mac/scoped_nsautorelease_pool.h"
#endif

#if !defined(OS_CHROMEOS) && defined(OS_LINUX)
#include "ui/base/ime/input_method_initializer.h"
#endif

namespace content {

ContentBrowserTest::ContentBrowserTest()
{
#if defined(OS_MACOSX)
    // See comment in InProcessBrowserTest::InProcessBrowserTest().
    base::FilePath content_shell_path;
    CHECK(PathService::Get(base::FILE_EXE, &content_shell_path));
    content_shell_path = content_shell_path.DirName();
    content_shell_path = content_shell_path.Append(
        FILE_PATH_LITERAL("Content Shell.app/Contents/MacOS/Content Shell"));
    CHECK(PathService::Override(base::FILE_EXE, content_shell_path));
#endif
    base::FilePath content_test_data(FILE_PATH_LITERAL("content/test/data"));
    CreateTestServer(content_test_data);
}

ContentBrowserTest::~ContentBrowserTest()
{
}

void ContentBrowserTest::SetUp()
{
    base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
    command_line->AppendSwitch(switches::kContentBrowserTest);

    SetUpCommandLine(command_line);

#if defined(OS_ANDROID)
    shell_main_delegate_.reset(new ShellMainDelegate);
    shell_main_delegate_->PreSandboxStartup();
    if (command_line->HasSwitch(switches::kSingleProcess)) {
        // We explicitly leak the new ContentRendererClient as we're
        // setting a global that may be used after ContentBrowserTest is
        // destroyed.
        ContentRendererClient* old_client = switches::IsRunLayoutTestSwitchPresent()
            ? SetRendererClientForTesting(new LayoutTestContentRendererClient)
            : SetRendererClientForTesting(new ShellContentRendererClient);
        // No-one should have set this value before we did.
        DCHECK(!old_client);
    }
#elif defined(OS_MACOSX)
    // See InProcessBrowserTest::PrepareTestCommandLine().
    base::FilePath subprocess_path;
    PathService::Get(base::FILE_EXE, &subprocess_path);
    subprocess_path = subprocess_path.DirName().DirName();
    DCHECK_EQ(subprocess_path.BaseName().value(), "Contents");
    subprocess_path = subprocess_path.Append(
        "Frameworks/Content Shell Helper.app/Contents/MacOS/Content Shell Helper");
    command_line->AppendSwitchPath(switches::kBrowserSubprocessPath,
        subprocess_path);
#endif

    // LinuxInputMethodContextFactory has to be initialized.
#if !defined(OS_CHROMEOS) && defined(OS_LINUX)
    ui::InitializeInputMethodForTesting();
#endif

    BrowserTestBase::SetUp();
}

void ContentBrowserTest::TearDown()
{
    BrowserTestBase::TearDown();

    // LinuxInputMethodContextFactory has to be shutdown.
#if !defined(OS_CHROMEOS) && defined(OS_LINUX)
    ui::ShutdownInputMethodForTesting();
#endif

#if defined(OS_ANDROID)
    shell_main_delegate_.reset();
#endif
}

void ContentBrowserTest::RunTestOnMainThreadLoop()
{
    if (!switches::IsRunLayoutTestSwitchPresent()) {
        CHECK_EQ(Shell::windows().size(), 1u);
        shell_ = Shell::windows()[0];
    }

#if defined(OS_MACOSX)
    // On Mac, without the following autorelease pool, code which is directly
    // executed (as opposed to executed inside a message loop) would autorelease
    // objects into a higher-level pool. This pool is not recycled in-sync with
    // the message loops' pools and causes problems with code relying on
    // deallocation via an autorelease pool (such as browser window closure and
    // browser shutdown). To avoid this, the following pool is recycled after each
    // time code is directly executed.
    base::mac::ScopedNSAutoreleasePool pool;
#endif

    // Pump startup related events.
    DCHECK(base::MessageLoopForUI::IsCurrent());
    base::RunLoop().RunUntilIdle();

#if defined(OS_MACOSX)
    pool.Recycle();
#endif

    SetUpOnMainThread();

    bool old_io_allowed_value = base::ThreadRestrictions::SetIOAllowed(false);
    RunTestOnMainThread();
    base::ThreadRestrictions::SetIOAllowed(old_io_allowed_value);

    TearDownOnMainThread();
#if defined(OS_MACOSX)
    pool.Recycle();
#endif

    for (RenderProcessHost::iterator i(RenderProcessHost::AllHostsIterator());
         !i.IsAtEnd(); i.Advance()) {
        i.GetCurrentValue()->FastShutdownIfPossible();
    }

    Shell::CloseAllWindows();
}

Shell* ContentBrowserTest::CreateBrowser()
{
    return Shell::CreateNewWindow(
        ShellContentBrowserClient::Get()->browser_context(),
        GURL(url::kAboutBlankURL),
        NULL,
        gfx::Size());
}

Shell* ContentBrowserTest::CreateOffTheRecordBrowser()
{
    return Shell::CreateNewWindow(
        ShellContentBrowserClient::Get()->off_the_record_browser_context(),
        GURL(url::kAboutBlankURL),
        NULL,
        gfx::Size());
}

} // namespace content
